package com.heima.wemedia.service.impl;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.heima.aliyun.GreenImageScan;
import com.heima.aliyun.GreenTextScan;
import com.heima.common.exception.CustException;
import com.heima.feigns.AdminFeign;
import com.heima.model.common.constants.message.PublishArticleConstants;
import com.heima.model.common.constants.wemedia.WemediaConstants;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.wemedia.pojos.ContentAndImage;
import com.heima.model.wemedia.pojos.WmNews;
import com.heima.utils.common.SensitiveWordUtil;
import com.heima.wemedia.mapper.WmNewsMapper;
import com.heima.wemedia.service.WmNewsAutoScanService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

@Service
@Slf4j
public class WmNewsAutoScanServiceImpl implements WmNewsAutoScanService {

    @Autowired
    GreenImageScan greenImageScan;
    @Autowired
    GreenTextScan greenTextScan;
    @Autowired
    AdminFeign adminFeign;
    @Autowired
    private WmNewsMapper wmNewsMapper;
    @Autowired
    RabbitTemplate rabbitTemplate;
    @Value("${file.oss.web-site}")
    String webSite;
    /**
     * 自媒体审核
     * @param wmNewsId  自媒体文章id
     */
    @Override
    public void autoScanWmNews(Integer wmNewsId) {
        //1. 根据文章id 远程调用feign查询文章
        if (wmNewsId == null) {
            log.error("自动审核文章失败    文章id为空");
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID);
        }
        WmNews wmNews = wmNewsMapper.selectById(wmNewsId);
        if (wmNews == null) {
            log.error("自动审核文章失败    未查询自媒体文章信息  wmNewsId:{}",wmNewsId);
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST);
        }
        // 2. 判断文章状态是否为待审核状态
        //问题：如果用short类型进行equals判断会不会报错？
        //结果：不会，正常执行
        if (WemediaConstants.WM_NEWS_SUMMIT_STATUS.equals(wmNews.getStatus())) {
            // 抽取出文章中 所有的文本内容 和 所有的图片url集合 Map<String,Object>  content 内容   images List<String>
            ContentAndImage contentAndImage = handleTextAndImages(wmNews);
            // 3.1  DFA敏感词审核    失败   修改文章状态(2)
            boolean flag = handleSensitive(contentAndImage.getContent(), wmNews);
            //DFA敏感词审核不通过直接返回
            if (!flag) {
                return;
            }
            log.info(" 自管理敏感词审核通过  =======   ");

            // 阿里云的文本审核   失败  状态2  不确定 状态3
            boolean isTextScan = handleTextScan(contentAndImage.getContent(),wmNews);
            if(!isTextScan) return;
            log.info(" 阿里云内容审核通过  =======   ");

            //阿里云审核图片
            List<String> images = contentAndImage.getImages();
            if (images != null) {
                boolean isImageScan =handleImageScan(images,wmNews);
                if(!isImageScan) {
                    return;
                }
                log.info(" 阿里云图片审核通过  =======   ");
            }
        }

        //审核通过就修改状态
        updateWmNews(wmNews,WemediaConstants.WM_NEWS_AUTHED_STATUS,"审核成功");
        // 通知定时发布文章
        long publishTime = wmNews.getPublishTime().getTime();
        // 当前时间
        long nowTime = new Date().getTime();
        long remainTime = publishTime - nowTime;
        // 发布文章
        rabbitTemplate.convertAndSend(PublishArticleConstants.DELAY_DIRECT_EXCHANGE
                , PublishArticleConstants.PUBLISH_ARTICLE_ROUTE_KEY
                , wmNews.getId()
                , (message)->{                              // 延时消息 必设置
                    message.getMessageProperties().setHeader("x-delay",remainTime<=0?0:remainTime);
                    return message;
                }
        );
        log.info("立即发布文章通知成功发送，文章id : {}", wmNews.getId());
    }

    /**
     * 抽取 文章中所有 文本内容  及 所有图片路径
     * @param wmNews 文章表
     * @return
     */
    private ContentAndImage handleTextAndImages(WmNews wmNews) {
        String contentJson = wmNews.getContent(); // [{},{},{}]
        if (StringUtils.isBlank(contentJson)) {
            log.error("自动审核文章失败    文章内容为空");
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID,"文章内容为空");
        }
        //每条文章中有内容和图片信息，即String String/List<String> 类型的集合
        List<Map> contentMaps = JSONArray.parseArray(contentJson, Map.class);
        // 1. 抽取文章中所有文本  抽取后的content为：家乡很美   _hmtt_   国家伟大
        String content = contentMaps.stream()
                .filter(map -> "text".equals(map.get("type")))
                .map(map -> (String) map.get("value"))
                .collect(Collectors.joining("_hmtt_"));
        content = content + "_hmtt_" + wmNews.getTitle();

        // 2. 抽取文章中所有图片   content :  全路径       images :  文件名称  + 访问前缀
        List<String> imageList = contentMaps.stream()
                .filter(map -> "image".equals(map.get("type")))
                .map(map -> (String) map.get("value"))
                .collect(Collectors.toList());
        if (StringUtils.isNotBlank(wmNews.getImages())) {
            // 按照 逗号 切割封面字符串  得到数组   基于数组得到stream   将每一条数据都拼接一个前缀 收集成集合
            List<String> urls = Arrays.stream(wmNews.getImages().split(","))
                    .map(url -> webSite + url)
                    .collect(Collectors.toList());
            imageList.addAll(urls);
        }
        // 3. 去重
        imageList = imageList.stream().distinct().collect(Collectors.toList());

//        Map result = new HashMap();
//        result.put("content",content);
//        result.put("images",imageList);
        //封装实体类
        ContentAndImage contentAndImage = new ContentAndImage();
        contentAndImage.setContent(content);
        contentAndImage.setImages(imageList);
        return contentAndImage;
    }

    /**
     * 阿里云文本判断
     * @param content
     * @param wmNews
     * @return
     */
    private boolean handleTextScan(String content, WmNews wmNews) {
        boolean flag = true;
        try {
            Map map = greenTextScan.greenTextScan(content);
            String suggestion = (String) map.get("suggestion");
            switch (suggestion) {
                case "block":
                    updateWmNews(wmNews,WmNews.Status.FAIL.getCode(),"文本中有违规内容，审核失败");
                    flag = false;
                    break;
                case "review":
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"文本中有不确定内容，转为人工审核");
                    flag = false;
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            log.error("阿里云文本审核出现异常 , 原因:{}",e.getMessage());
            updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"阿里云内容服务异常，转为人工审核");
            flag = false;
        }
        return flag;
    }

    /**
     * 阿里云检测敏感内容
     * @param images 图片集合
     * @param wmNews 文章表
     * @return
     */
    private boolean handleImageScan(List<String> images, WmNews wmNews) {
        boolean flag = true;
        try {
            Map map = greenImageScan.imageUrlScan(images);
            String suggestion = (String) map.get("suggestion");
            switch (suggestion) {
                case "block":
                    updateWmNews(wmNews,WemediaConstants.WM_NEWS_AUTH_FAIL,"内容违规");
                    flag = false;
                    break;
                case "review":
                    updateWmNews(wmNews,WemediaConstants.ADMIN_AUTH,"转为人工审核");
                    flag = false;
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            log.error("阿里云图片审核出现异常 , 原因:{}",e.getMessage());
            updateWmNews(wmNews,WemediaConstants.ADMIN_AUTH,"转为人工审核");
            flag = false;
        }
        return flag;
    }

    /**
     * 基于DFA检测内容（自定义）敏感词，先于阿里云检测执行
     * @param content 敏感词
     * @param wmNews 文章表
     * @return
     */
    private boolean handleSensitive(String content,WmNews wmNews) {
        boolean flag = true;
        //从敏感词表中查询所有敏感词，远程调用feign
        //注意要给泛型，不然会报错
        ResponseResult<List<String>> responseResult = adminFeign.sensitives();
        List<String> sensitivesList = responseResult.getData();
        //使用DFA工具类
        SensitiveWordUtil.initMap(sensitivesList);
        //检测敏感词
        Map<String, Integer> matchWords = SensitiveWordUtil.matchWords(content);
        //判断是否有
        if (matchWords != null && matchWords.size() > 0) {
            //有更改状态为 2 - 审核失败
            updateWmNews(wmNews,WemediaConstants.WM_NEWS_AUTH_FAIL,"出现敏感信息");
            flag = false;
            return flag;
        }
        return flag;
    }

    /**
     * 修改文章状态
     * 因为要多次用到修改状态的方法，所有提取
     * @param wmNews 文章表
     * @param status 状态
     * @param reason 原因
     */
    private void updateWmNews(WmNews wmNews, short status, String reason) {
        wmNews.setStatus(status);
        wmNews.setReason(reason);
        wmNewsMapper.updateById(wmNews);
    }

}
